from configparser import ConfigParser
from unittest.mock import MagicMock

import pytest
from langchain_core.messages import BaseMessage, SystemMessage, HumanMessage

from llm_client_redis.llm_client import LLMClientRedis


@pytest.fixture
def llm_client() -> LLMClientRedis:
    return LLMClientRedis()


def test_request(llm_client: LLMClientRedis):

    model: str = "home_deepseek-r1:32b"

    messages: list[BaseMessage] = [SystemMessage("你是一个好助手"), HumanMessage("你好")]

    data = llm_client.request(messages=messages, model=model)

    print(data)

    assert data is not None




